<?php

namespace Rsa;
/**
 * RSA加密算法
 * RSA算法类
 * 签名及密文编码：base64字符串/十六进制字符串/二进制字符串流
 * 填充方式: PKCS1Padding（加解密）/NOPadding（解密）
 * Notice:Only accepts a single block. Block size is equal to the RSA key size!
 * 如密钥长度为1024 bit，则加密时数据需小于128字节，加上PKCS1Padding本身的11字节信息，所以明文需小于117字节
 * +-----------------------------
 * Company: 170
 * +-----------------------------
 * Date: 2019/11/20
 * +-----------------------------
 * Time: 15:37
 * +-----------------------------
 * Created by PhpStorm.
 * +-----------------------------
 */
use think\facade\env;
class ResponseRSA
{
    /**
     * 公钥（文件类）
     * @var null
     */
    private $pubKey = null;

    /**
     * 私钥（文件类）
     * @var null
     */
    private $priKey = null;

    /**
     * 构造函数
     * @param string 公钥文件（验签和加密时传入）
     * @param string 私钥文件（签名和解密时传入）
     */
    public function __construct()
    {

        $public_key_file = env::get('extend_path') ."Rsa/public_key_return_pro.txt";

        if ($public_key_file) {
            $this->_getPublicKey($public_key_file);
        }
         $private_key_file = env::get('extend_path') ."Rsa/private_key_return_pro.txt";
        if ($private_key_file) {
            $this->_getPrivateKey($private_key_file);
        }
    }

    /**
     * 自定义错误处理
     * @param $msg 错误信息
     * @return bool
     */
    private function _error($msg)
    {
        die('RSA Error:' . $msg); //TODO
    }

    /**
     * 检测填充类型
     * 加密只支持PKCS1_PADDING
     * 解密支持PKCS1_PADDING和NO_PADDING
     * @param int 填充模式
     * @param string 加密en/解密de
     * @return bool
     */
    private function _checkPadding($padding, $type)
    {
        if ($type == 'en') {
            switch ($padding) {
                case OPENSSL_PKCS1_PADDING:
                    $ret = true;
                    break;
                default:
                    $ret = false;
            }
        } else {
            switch ($padding) {
                case OPENSSL_PKCS1_PADDING:
                    $ret = true;
                    break;
                case OPENSSL_NO_PADDING:
                    $ret = true;
                    break;
                default:
                    $ret = false;
            }
        }
        return $ret;
    }

    /**
     * 验证初始加密数据
     * @param $data 加密数据
     * @param $code 加密方式
     * @return string
     */
    private function _encode($data, $code)
    {

        switch (strtolower($code)) {
            case 'base64':
                $data = base64_encode('' . $data);
                break;
            case 'hex':
                $data = bin2hex($data);
                break;
            default:
                $data = base64_encode('' . $data);
        }
        return $data;
    }

    /**
     * 初始解密数据
     * @param $data 解密数据
     * @param $code 加密方式
     * @return bool|string
     */
    private function _decode($data, $code)
    {
        switch (strtolower($code)) {
            case 'base64':
                $data = base64_decode($data);
                break;
            case 'hex':
                $data = $this->_hex2bin($data);
                break;
            default:
                $data = base64_encode('' . $data);
        }

        return $data;
    }


    /**
     * 读取公钥文件内容
     * @param $file 公钥文件地址
     */
    private function _getPublicKey($file)
    {
        $key_content = $this->_readFile($file);
        //echo $key_content;exit;
        if ($key_content) {
            $this->pubKey = openssl_get_publickey($key_content);
        }
        // echo $this->pubKey;exit;
    }

    /**
     * 读取公钥文件内容
     * @param $file 私钥文件地址
     */
    private function _getPrivateKey($file)
    {

        $key_content = $this->_readFile($file);

        if ($key_content) {
            $this->priKey = openssl_get_privatekey($key_content);
        }
    }

    /**
     * 打开并读取文件内容
     * @param $file 验证文件地址并读取文件内容
     * @return bool|string  返回结果
     */
    private function _readFile($file)
    {
        $ret = false;
        if (!file_exists($file)) {
            $this->_error("The file {$file} is not exists");
        } else {
            $ret = file_get_contents($file);
        }
        return $ret;
    }

    /**
     * 把十六进制值转换为 ASCII 字符
     * @param bool $hex
     * @return bool|string
     */
    private function _hex2bin($hex = false)
    {
        $ret = $hex !== false && preg_match('/^[0-9a-fA-F]+$/i', $hex) ? pack("H*", $hex) : false;
        return $ret;
    }

    /**
     * 生成签名(私钥创签)
     * @param string 签名材料
     * @param string 签名编码（base64/hex/bin）
     * @param int $padding 填充方式（貌似php有bug，所以目前仅支持OPENSSL_PKCS1_PADDING）
     * @return 签名值
     */
    public function sign($data, $code = 'base64', $padding = OPENSSL_ALGO_MD5)
    {
        $ret = false;
        openssl_sign($data, $ret, $this->priKey, $padding);

        if (openssl_sign($data, $ret, $this->priKey, $padding)) {
            $ret = $this->_encode($ret, $code);
        }

        return $ret;
    }

    /**
     * 验证签名(公钥验签)
     * @param string 签名材料
     * @param string 签名值
     * @param string 签名编码（base64/hex/bin）
     * @param int $padding 填充方式（貌似php有bug，所以目前仅支持OPENSSL_PKCS1_PADDING）
     * @return bool
     */
    public function verify($data, $sign, $code = 'base64', $padding = OPENSSL_ALGO_MD5)
    {
        $ret = false;
        $sign = $this->_decode($sign, $code);

        if ($sign !== false) {
            $ret = (bool)openssl_verify($data, $sign, $this->pubKey, $padding);
            openssl_free_key($this->pubKey); // 释放资源
        }
        return $ret;
    }

    /**
     * 加密
     * @param string 明文
     * @param string 密文编码（base64/hex/bin）
     * @param int $padding 填充方式（貌似php有bug，所以目前仅支持OPENSSL_PKCS1_PADDING）
     * @return string 密文
     */
    public function encrypt($data, $code = 'base64', $padding = OPENSSL_PKCS1_PADDING)
    {
        $ret = false;
        if (!$this->_checkPadding($padding, 'en')) $this->_error('padding error');
        if (openssl_public_encrypt($data, $result, $this->pubKey, $padding)) {
            $ret = $this->_encode($result, $code);
        }
        return $ret;
    }

    public function encryptss($data, $code = 'base64', $padding = OPENSSL_PKCS1_PADDING)
    {
        $ret = false;
        if (!$this->_checkPadding($padding, 'en')) $this->_error('padding error');
        $crypto = '';
        foreach (str_split($data,117) as $value) {

            if (openssl_public_encrypt($value, $result, $this->pubKey, $padding)) {
                $crypto .= $result;
            }
        }
        $ret = $this->_encode($crypto, $code);
       
        return $ret;
    }

    /**
     * 解密
     * @param string 密文
     * @param string 密文编码（base64/hex/bin）
     * @param int 填充方式（OPENSSL_PKCS1_PADDING / OPENSSL_NO_PADDING）
     * @param bool 是否翻转明文（When passing Microsoft CryptoAPI-generated RSA cyphertext, revert the bytes in the block）
     * @return string 明文
     */
    public function decrypt($data, $code = 'base64', $padding = OPENSSL_PKCS1_PADDING, $rev = false)
    {
        $ret = false;
        $data = $this->_decode($data, $code);

        if (!$this->_checkPadding($padding, 'de')) $this->_error('padding error');
        if ($data !== false) {
            if (openssl_private_decrypt($data, $result, $this->priKey, $padding)) {
                $ret = $rev ? rtrim(strrev($result), "\0") : '' . $result;
            }
        }
        return $ret;
    }

    public function decryptss($data, $code = 'base64', $padding = OPENSSL_PKCS1_PADDING, $rev = false)
    {
        $ret = false;
        $data = $this->_decode($data, $code);

        if (!$this->_checkPadding($padding, 'de')) $this->_error('padding error');
        if ($data !== false) {
             $crypto = '';
            foreach (str_split($data,128) as $value) {
                openssl_private_decrypt($value, $result, $this->priKey, $padding);
                $crypto .= $result;
                
            }
            $ret = $rev ? rtrim(strrev($crypto), "\0") : '' . $crypto;
            
        }
        return $ret;
    }


    //验证签名
    public function getMd5($array)
    {
        if(empty($array)){
            return false;
        }
        $info = $array;
        unset($info['signStr']);
        $string = '';
        ksort($info);
        foreach ($info as $k => $v) {
            $string .=$v;
        }
        $signKey = config('env.signKey'); 
        $string.= $signKey;
        $signStr = md5($string);
        if($signStr == $array['signStr']){
            return true;
        }else{
            return false;
        }

    }
    public function setMd5($array)
    {
        $signKey = config('env.signKey'); 
        $string ='';
        ksort($array);
        foreach ($array as $k => $v) {
            $string .=$v;
        }
        $string.= $signKey;
        return md5($string);
    }


}